//给你一个字符串 s，请你将 s 分割成一些子串，使每个子串都是 回文串 。返回 s 所有可能的分割方案。 
//
// 回文串 是正着读和反着读都一样的字符串。 
//
// 
//
// 示例 1： 
//
// 
//输入：s = "aab"
//输出：[["a","a","b"],["aa","b"]]
// 
//
// 示例 2： 
//
// 
//输入：s = "a"
//输出：[["a"]]
// 
//
// 
//
// 提示： 
//
// 
// 1 <= s.length <= 16 
// s 仅由小写英文字母组成 
// 
//
// Related Topics 字符串 动态规划 回溯 👍 1616 👎 0


package LeetCode.editor.cn;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * @author ldltd
 * @date 2023-09-07 17:14:16
 * @description 131.分割回文串
 */
public class PalindromePartitioning {
	public static void main(String[] args) {
		//测试代码
		Solution solution = new PalindromePartitioning().new Solution();
		System.out.println(solution.partition("aab"));
	}

	//力扣代码
//leetcode submit region begin(Prohibit modification and deletion)
	class Solution {
		/*回溯+动态规划预处理回文串*/
		public List<List<String>> partition(String s) {
			List<List<String>> res = new ArrayList<>();
			int n = s.length();
			boolean[][] dp = new boolean[n][n];
//		初始化，所有长度为1的，都是回文
			Arrays.setAll(dp, i -> {
				boolean[] row = new boolean[n];
				row[i] = true;
//				res.add(new ArrayList<>(Collections.singletonList(s.charAt(i))));
				return row;
			});
			char[] chars = s.toCharArray();
//其他写法
//			for (int i = n - 1; i >= 0; --i) {
//				for (int j = i + 1; j < n; ++j) {
//					dp[i][j] = (s.charAt(i) == s.charAt(j)) && dp[i + 1][j - 1];
//				}
//			}
//

			//从两个长度开始枚举
			for (int len = 2; len <= n; len++) {
				for (int l = 0; l < n; l++) {
					//由l-r+1=L,得出右边界
					int r = len + l - 1;
					//越界退出
					if (r >= n) {
						break;
					}
					//赋值
					if (chars[l] != chars[r]) {
						dp[l][r] = false;
					} else {
						//相等，且间隔一个 比如 012  2-0=2 中间只有一个
						if (r - l < 3) {
							dp[l][r] = true;
						} else {
							//dp[l][r]==true等价于char[l]==char[r]且dp[l+1]==dp[r-1]
							dp[l][r] = dp[l + 1][r - 1];
						}
					}
					//遍历完，如果dp[i][L]==true,保存
//				if(dp[l][r]){
//					res.add(new ArrayList<>())
//				}
				}
			}
			backtrack(s, 0, dp, new ArrayList<>(), res);
			return res;
		}

		private void backtrack(String s, int start, boolean[][] dp, List<String> path, List<List<String>> res) {
			if (start == s.length()) {
				res.add(new ArrayList<>(path));
				return ;
			}
			//从start开始，从0~start的字符串已经分割好放入path中了，
			//start到end中的不是回文的直接加入，end+1的下一个位置作为新的回溯位置
			for (int end = start; end < s.length(); end++) {
				//每个单个都是回文
				if (dp[start][end]) {
					path.add(s.substring(start, end + 1));
					backtrack(s, end + 1, dp, path, res);
					path.remove(path.size() - 1);
				}
			}
		}

		/*把预处理变成记忆化搜索*/
		int [][]f;
		List<List<String >> res=new ArrayList<>();
		List<String>ans=new ArrayList<>();
		int n;
		public List<List<String>> partition1(String s){
			n=s.length();
			f=new int[n][n];
			dfs(s,0);
			return res;
		}
		private void dfs(String s,int i){
			if(i==n){
				res.add(new ArrayList<>(ans));
				return;
			}
			for (int j = i; j <n ; j++) {
				if(isPalindrome(s,i,j)==1){
					ans.add(s.substring(i,j+1));
					dfs(s,j+1);
					ans.remove(ans.size()-1);
				}
			}
		}
		//记忆化搜索，f[i][j]=0表示没搜，1表示是回文，-1表示不是回文
		private int isPalindrome(String s,int i,int j){
			//如果已经搜搜过了就直接返回
			if(f[i][j]!=0){
				return f[i][j];
			}
			//因为是顺序的，所以从0开始的左边已经赋值了 0到3是回文，则3到0也是回文
			if(i>=j){
				f[i][j]=1;
			}else if(s.charAt(i)==s.charAt(j)){
				f[i][j]=isPalindrome(s,i+1,j-1);
			}else {
				f[i][j]=-1;
			}
			return f[i][j];
		}
	}
//leetcode submit region end(Prohibit modification and deletion)

	}

